package model;

import java.util.ArrayList;
import java.util.List;

public abstract class Subtreatment {

//	private int id;
	private String name;
	private List<StartStop> startstops;

	public Subtreatment(String name) {
		this.name = name;
//		setId();
		startstops = new ArrayList<StartStop>();
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}
	
//	public int getId(){
//		return id;
//	}
//	
//	public void setId(){
//		if(this instanceof Coating){
//			id = 1;
//		} else if (this instanceof DryTime){
//			id = 2;
//		}
//	}
//	
	public String toString() {
		return name;
	}
	

	
	public void addStartStop(StartStop startstop){
		if(startstop !=null && startstop.getSubtreatment()!=null)
			startstop.getSubtreatment().removeStartStop(startstop);
		if(!startstops.contains(startstop)){
			startstops.add(startstop);
			startstop.setSubtreatment(this);
		}
	}
	
	private void removeStartStop(StartStop startstop){
		if(startstops.contains(startstop)){
			startstops.remove(startstop);
			startstop.setSubtreatment(null);
		}
	}
	
	public ArrayList<StartStop> getStartStop(){
		return new ArrayList<StartStop>(startstops);
	}	

}
